package io.renren.modules.health.controller;

import io.renren.common.annotation.LogOperation;
import io.renren.common.constant.Constant;
import io.renren.common.exception.RenException;
import io.renren.common.page.PageData;
import io.renren.common.utils.ExcelUtils;
import io.renren.common.utils.Result;
import io.renren.common.validator.AssertUtils;
import io.renren.common.validator.ValidatorUtils;
import io.renren.common.validator.group.AddGroup;
import io.renren.common.validator.group.DefaultGroup;
import io.renren.common.validator.group.UpdateGroup;
import io.renren.modules.health.dto.FoodDTO;
import io.renren.modules.health.dto.FoodPlanDetailDTO;
import io.renren.modules.health.dto.FoodPrePlanDTO;
import io.renren.modules.health.dto.FoodPrePlanDetailDTO;
import io.renren.modules.health.entity.FoodPrePlanDetailEntity;
import io.renren.modules.health.excel.FoodPrePlanExcel;
import io.renren.modules.health.service.FoodPrePlanDetailService;
import io.renren.modules.health.service.FoodPrePlanService;
import io.renren.modules.health.service.FoodService;
import io.renren.modules.sys.entity.SysUserEntity;
import io.renren.modules.sys.service.SysUserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletResponse;
import java.util.*;
import java.util.stream.Collectors;


/**
 * 饮食方案
 */
@RestController
@RequestMapping("health/foodpreplan")
@Api(tags="饮食方案")
public class FoodPrePlanController {
    @Autowired
    private FoodPrePlanService foodPrePlanService;
    @Autowired
    private FoodPrePlanDetailService foodPrePlanDetailService;
    @Autowired
    private SysUserService sysUserService;
    @Autowired
    private FoodService foodService;

    @GetMapping("page")
    @ApiOperation("分页")
    @ApiImplicitParams({
        @ApiImplicitParam(name = Constant.PAGE, value = "当前页码，从1开始", paramType = "query", required = true, dataType="int") ,
        @ApiImplicitParam(name = Constant.LIMIT, value = "每页显示记录数", paramType = "query",required = true, dataType="int") ,
        @ApiImplicitParam(name = Constant.ORDER_FIELD, value = "排序字段", paramType = "query", dataType="String") ,
        @ApiImplicitParam(name = Constant.ORDER, value = "排序方式，可选值(asc、desc)", paramType = "query", dataType="String")
    })
    @RequiresPermissions("health:foodpreplan:page")
    public Result<PageData<FoodPrePlanDTO>> page(@ApiIgnore @RequestParam Map<String, Object> params){
        PageData<FoodPrePlanDTO> page = foodPrePlanService.page(params);
        if (page.getTotal() > 0) {
            List<Long> creators = page.getList().stream().map(FoodPrePlanDTO::getCreator).collect(Collectors.toList());
            List<Long> updaters = page.getList().stream().map(FoodPrePlanDTO::getUpdater).collect(Collectors.toList());
            List<Long> userIds = new ArrayList<>(10);
            userIds.addAll(creators);
            userIds.addAll(updaters);
            userIds = new ArrayList<>(new HashSet<>(userIds));
            List<SysUserEntity> users = sysUserService.getUsersByIds(userIds);
            page.getList().forEach(l -> {
                l.setCreatorName(users.stream().filter(u -> u.getId().equals(l.getCreator())).findFirst().orElse(new SysUserEntity()).getUsername());
                l.setUpdaterName(users.stream().filter(u -> u.getId().equals(l.getUpdater())).findFirst().orElse(new SysUserEntity()).getUsername());
            });
        }
        return new Result<PageData<FoodPrePlanDTO>>().ok(page);
    }

    @GetMapping("{id}")
    @ApiOperation("信息")
    @RequiresPermissions("health:foodpreplan:info")
    public Result<FoodPrePlanDTO> get(@PathVariable("id") Long id){
        FoodPrePlanDTO data = foodPrePlanService.get(id);
        HashMap<String, Object> param = new HashMap<>();
        param.put("foodPrePlanId",id);
        List<FoodPrePlanDetailDTO> foodPrePlanDetailDTOS = foodPrePlanDetailService.list(param);
        param.put("foodIds",foodPrePlanDetailDTOS.stream().map(FoodPrePlanDetailDTO::getFoodId).distinct().collect(Collectors.toList()));
        data.setFoodPrePlanDetails(foodPrePlanDetailDTOS);
        List<FoodDTO> foodDTOS = foodService.list(param);
        foodPrePlanDetailDTOS.forEach(foodPrePlanDetail -> foodPrePlanDetail.setFoodName(
                foodDTOS.stream().filter(food -> foodPrePlanDetail.getFoodId().equals(food.getId()))
                        .findFirst()
                        .orElse(new FoodDTO()).getFoodName()));

        return new Result<FoodPrePlanDTO>().ok(data);
    }

    @PostMapping
    @ApiOperation("保存")
    @LogOperation("保存")
    @RequiresPermissions("health:foodpreplan:save")
    public Result save(@RequestBody FoodPrePlanDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, AddGroup.class, DefaultGroup.class);

        foodPrePlanService.save(dto);
        if (!Objects.isNull(dto.getFoodPrePlanDetails()) && !dto.getFoodPrePlanDetails().isEmpty()) {
            List<FoodPrePlanDetailEntity> foodPrePlanDetailEntities = dto.getFoodPrePlanDetails().stream()
                    .map(fpd -> {
                        fpd.setFoodPrePlanId(dto.getId());
                        FoodPrePlanDetailEntity foodPrePlanEntity = new FoodPrePlanDetailEntity();
                        BeanUtils.copyProperties(fpd, foodPrePlanEntity);
                        return foodPrePlanEntity;
                    })
                    .collect(Collectors.toList());
            foodPrePlanDetailService.insertBatch(foodPrePlanDetailEntities);
        } else throw new RenException(100,"饮食方案未添加食品");
        return new Result();
    }

    @PutMapping
    @ApiOperation("修改")
    @LogOperation("修改")
    @RequiresPermissions("health:foodpreplan:update")
    public Result update(@RequestBody FoodPrePlanDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, UpdateGroup.class, DefaultGroup.class);

        foodPrePlanService.update(dto);

        return new Result();
    }

    @DeleteMapping
    @ApiOperation("删除")
    @LogOperation("删除")
    @RequiresPermissions("health:foodpreplan:delete")
    public Result delete(@RequestBody Long[] ids){
        //效验数据
        AssertUtils.isArrayEmpty(ids, "id");

        foodPrePlanService.delete(ids);
        HashMap<String, Object> param = new HashMap<>();
        param.put("foodPrePlanIds", Arrays.asList(ids));
        List<FoodPrePlanDetailDTO> planPreDetailDTOList = foodPrePlanDetailService.list(param);
        foodPrePlanDetailService.deleteBatchIds(planPreDetailDTOList.stream().map(FoodPrePlanDetailDTO::getFoodId).collect(Collectors.toList()));
        return new Result();
    }

    @GetMapping("export")
    @ApiOperation("导出")
    @LogOperation("导出")
    @RequiresPermissions("health:foodpreplan:export")
    public void export(@ApiIgnore @RequestParam Map<String, Object> params, HttpServletResponse response) throws Exception {
        List<FoodPrePlanDTO> list = foodPrePlanService.list(params);

        ExcelUtils.exportExcelToTarget(response, null, "饮食方案", list, FoodPrePlanExcel.class);
    }

}